#!/usr/bin/bash




#trainset=/home/gaoxinglong/env/tools/speech_tools/kaldi/egs/librispeech/us48_s5/exp/tri5b_ali_a_c/ali2pdfs/chunk_feats_labels
#trainset=/home/gaoxinglong/env/tools/speech_tools/kaldi/egs/librispeech/us48_s5/exp/tri4b_ali_72w-score9595s/ali2pdfs/chunk_feats_labels
#trainset=/home/gaoxinglong/env/tools/speech_tools/kaldi/egs/librispeech/us48_s5/exp/tri5b_ali_300w-score8585//ali2pdfs/chunk_feats_labels
#trainset=/home/gaoxinglong/env/tools/speech_tools/kaldi/egs/librispeech/us48_s5/exp/tri4b_ali_native_58hus58h/ali2pdfs/chunk_feats_labels
#trainset=/home/gaoxinglong/env/tools/speech_tools/kaldi/egs/librispeech/us48_s5/exp/tri4b_ali_native_58hus58h/ali2pdfs/
#trainset=/home/gaoxinglong/env/tools/speech_tools/kaldi/egs/librispeech/us48_s5/exp/tri5b_ali_gop_dataus73h/ali2pdfs/gop_feats/
#trainset=/ai-cephfs/Share/xlg_share/anyword_experiment/from600w_lt70//ali2pdfs/feats_and_state_labels
trainset=/ai-cephfs/Share/xlg_share/anyword_experiment/tri6b_ali_from1020w_lt70kid1234h/ali2pdfs/feats_and_state_labels
#trainset=/ai-cephfs/Share/xlg_share/anyword_experiment/tri6b_ali_from1020w_lt70kid1234h/torchfeatures/feats_and_state_labels
#trainset=/home/gaoxinglong/env/tools/speech_tools/kaldi/egs/librispeech/us48_s5/exp/tri5b_ali_train_2kshort85/ali2pdfs/chunk_feats_labels
#trainset=/home/gaoxinglong/env/tools/speech_tools/kaldi/egs/librispeech/us48_s5/exp/tri5b_ali_180w-score8585/ali2pdfs/chunk_feats_labels
feats=$trainset/final_feats_scps.list
labels=$trainset/final_labels_scps.list


batch_size=64
start_lr=0.0017
final_lr=0.00017
#num_labels=6112
#num_labels=3856
#num_labels=5376
num_labels=5688
context=14
epochs=20
#devices="0,1,2,3"
#devices="4,5,6,7"
#devices="0,1,6,7"
devices="0,1,2,3,4,5,6,7"
model_dir=kid1234h_am3
restart_model=
export CUDA_VISIBLE_DEVICES=$devices

python train-state-on-frames-apex.py --gpu_divices=$devices --epoch=$epochs --context_width=$context \
--feats_scp_list=$feats --labels_scp_list=$labels --save_dir=$model_dir \
--batch_size=$batch_size --output_num=$num_labels --start_lr=$start_lr --final_lr=$final_lr --restart_model=$restart_model

echo 'training done.'

